//package com.auth.utils;
//
//import com.auth.pojo.dto.SecurityUserDTO;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.security.oauth2.common.DefaultOAuth2AccessToken;
//import org.springframework.security.oauth2.common.OAuth2AccessToken;
//import org.springframework.security.oauth2.provider.OAuth2Authentication;
//import org.springframework.security.oauth2.provider.token.TokenEnhancer;
//import org.springframework.stereotype.Component;
//
//import java.util.HashMap;
//import java.util.Map;
//
///**
// * JWT内容增强器
// *
// * author wangbaohai
// */
//@Component
//@Slf4j
//public class JwtTokenEnhancer implements TokenEnhancer {
//    @Override
//    public OAuth2AccessToken enhance(OAuth2AccessToken accessToken, OAuth2Authentication authentication) {
//        SecurityUserDTO securityUser = (SecurityUserDTO) authentication.getPrincipal();
//        log.info("securityUser:{}",securityUser);
//        Map<String, Object> info = new HashMap<>();
//        //把用户ID设置到JWT中
//        info.put("userId", securityUser.getId());
//        info.put("userName", securityUser.getUsername());
//        info.put("userCode", securityUser.getUserCode());
//        ((DefaultOAuth2AccessToken) accessToken).setAdditionalInformation(info);
//        return accessToken;
//    }
//}
